Tensors and Dynamic neural networks in Python with strong GPU acceleration

Deezer source separation library including pretrained models.

Keras 3 is a multi-backend deep learning framework, with support for JAX, TensorFlow, PyTorch, and OpenVINO (for inference-only). Effortlessly build and train models for computer vision, natural language processing, audio processing, timeseries forecasting, recommender systems, etc.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.